package com.lagou.edu.utils;

import java.io.FileInputStream;
import java.io.IOException;
import java.io.OutputStream;

/**
 * @author 赵虎
 * @date 2020/11/8 6:00 下午
 */
public class StaticResourceUtil {

    public static String getAbsolutePath(String path) {
        String absolutePath = StaticResourceUtil.class.getResource("/").getPath();
        return absolutePath.replaceAll("\\\\", "/") + path;
    }

    public static void outputStaticResource(FileInputStream inputStream, OutputStream outputStream) throws IOException {
        int count = 0;
        while (count == 0) {
            count = inputStream.available();
        }
        int resourceSize = count;

        outputStream.write(HttpProtocolUtil.getHttpHeader200(resourceSize).getBytes());

        long written = 0;
        int byteSize = 1024;
        byte[] bytes = new byte[byteSize];

        while (written < resourceSize) {
            if (written + byteSize > resourceSize) {
                byteSize = (int) (resourceSize - written);
                bytes = new byte[byteSize];
            }
            inputStream.read(bytes);
            outputStream.write(bytes);
            outputStream.flush();
            written += byteSize;
        }
    }
}
